﻿#include "download.h"
#include "TimeBase.h"
#include <string>
#include <cstdlib>

int main(int argc, char* argv[]) {
    if (argc != 3) {
        std::cerr << "使用方法: 程序名 <URL> <文件名>" << std::endl;
        return 1;
    }

    std::string url = argv[1];
    std::string filename = argv[2];
    {
        TimeBase main_timer("下载文件" + filename);

        // URL 校验（基本校验，可以扩展）
        if (url.empty() || url.find("http") != 0) {
            std::cerr << "无效的 URL 格式" << std::endl;
            return 1;
        }

        // 下载文件
        if (download_file(url, filename) == 0) {
            std::cout << "文件已成功下载为 " << filename << std::endl;
        }
        else {
            std::cerr << "下载失败，请检查错误信息。" << std::endl;
        }
    }
    system("pause");
    return 0;
}
